# 导入shutil模块和os模块
import pandas as pd
import os
import glob
import re
import time
# from program.pack.append_df_to_excel import append_df_to_excel
import shutil, os
import numpy as np
import warnings

# 忽略警告
warnings.filterwarnings("ignore")

# 列显示不全，进行设置
pd.set_option('display.max_columns', 500)
pd.set_option('display.unicode.ambiguous_as_wide', True)
pd.set_option('display.unicode.east_asian_width', True)
pd.set_option('display.width', 180)  # 设置打印宽度(**重要**)

path_input = r'I:\TAI-40-II\2022-01-19-2#-jichang'
path_out = r'W:\wangshuan\TAI40-II\train'
for root_dirs, sub_dirs, files in os.walk(path_input):
    if 'Data' in sub_dirs:
        sub_dirs.remove('Data')
        for sub_dir in sub_dirs:  # 复制文件夹
            files_name = os.path.join(root_dirs, sub_dir)
            files_name_new = path_out + '\\' + files_name.split('\\')[-2] + '\\' + files_name.split('\\')[-1]
            print(files_name)
            print(files_name_new)
            try:
                shutil.copytree(files_name, files_name_new)
            except:
                pass
        for file in files:  # 复制文件
            file_name_1 = os.path.join(root_dirs, file)
            file_name_new_1 = path_out + '\\' + file_name_1.split('\\')[-2] + '\\' + file_name_1.split('\\')[-1]
            print(file_name_1)
            print(file_name_new_1)
            try:
                shutil.copy(file_name_1, file_name_new_1)
            except:
                pass


'''
#文件、文件夹的移动、复制、删除、重命名

#导入shutil模块和os模块
import shutil,os

#复制单个文件
shutil.copy("C:\\a\\1.txt","C:\\b")
#复制并重命名新文件
shutil.copy("C:\\a\\2.txt","C:\\b\\121.txt")
#复制整个目录(备份)
shutil.copytree("C:\\a","C:\\b\\new_a")

#删除文件
os.unlink("C:\\b\\1.txt")
os.unlink("C:\\b\\121.txt")
#删除空文件夹
try:
    os.rmdir("C:\\b\\new_a")
except Exception as ex:
    print("错误信息："+str(ex))#提示：错误信息，目录不是空的
#删除文件夹及内容
shutil.rmtree("C:\\b\\new_a")

#移动文件
shutil.move("C:\\a\\1.txt","C:\\b")
#移动文件夹
shutil.move("C:\\a\\c","C:\\b")

#重命名文件
shutil.move("C:\\a\\2.txt","C:\\a\\new2.txt")
#重命名文件夹
shutil.move("C:\\a\\d","C:\\a\\new_d")
'''
